#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@NAME          : route_response.py
@TIME          : 2025/03/09 20:52:52
@AUTHOR        : chenlip
@VERSION       : 0.0.1
@DESCRIPTION   : 统一路由响应处理
'''
# --------------------------引用--------------------------
import os
from flask import jsonify
from app.common.error_codes import ErrorCode
# --------------------------引用--------------------------

class RouterResponse:
    '''
    统一路由响应处理
    '''
    @staticmethod
    def success(message="操作成功", data=None):
        return ErrorCode.success_response(data=data, message=message)
    
    @staticmethod
    def error(code="10001", message=None, data=None):
        if not message:
            message = ErrorCode.get_message(code)
        return ErrorCode.error_response(code=code, message=message, data=data)
    

class ApiResponse:
    '''
    统一API响应处理, 用于返回json数据, 
    注意data数据类型, 一般为dict, list, str, int, float等, 必须是json序列化的数据
    '''
    @staticmethod
    def success(message="操作成功", data=None):
        _retn = ErrorCode.success_response(data=data, message=message)
        return jsonify(_retn)
    
    @staticmethod
    def error(code="10001", message=None, data=None):
        if not message:
            message = ErrorCode.get_message(code)
        _retn = ErrorCode.error_response(code=code, message=message, data=data)
        return jsonify(_retn)